tensorflow>=2.5.0
numpy
absl-py==0.10.0
tensorflow_datasets
ml_collections
